import pymysql
from config.project_baseinfo import *
class DBUtil:
    def __init__(self):
        self.db = pymysql.connect(**db_info)
        self.cursor = self.db.cursor()#创建游标初始化

    def query(self,sql):#查询
        self.cursor.execute(sql)#执行查询结果
        r = self.cursor.fetchall()#查看结果---返回元组
        result = []#初始化列表
        #遍历一边数据，然后加到字典里边，最后加入到列表里边
        for i in r:
            r_d={}
            for index,column in enumerate(self.cursor.description):
                r_d[column[0]] = i[index]
            result.append(r_d)#加入列表操作
        return result#[{id:1,name:mamou}]

    def excute(self,sql):#执行ddl
        self.cursor.execute(sql)#执行sql语句
        self.db.commit()#提交

    def close_db(self):#关闭连接
        self.cursor.close()
        self.db.close()